from decision_tree.Tree import CartTree
from utils.DataUtil import DataUtil

path = "/home/xiefeihong/PycharmProjects/SimpleMachineLearning/static/classify/{}.txt".format("mushroom")
train_num = 6000
x, y = DataUtil.get_dataset("mushroom", path, tar_idx=0, n_train=train_num)
tree = CartTree()
tree.fit(x, y, train_only=True)
